#!/bin/bash

# 创建存储图片和XML文件的目录
mkdir -p /home/data/allData/images
mkdir -p /home/data/allData/xml

# 遍历当前目录及其子目录下的所有jpg图片文件
find /home/data/ -type f -name "*.jpg" -exec cp {} /home/data/allData/images/ \;

# 复制对应的XML文件
find /home/data/ -type f -name "*.xml" -exec cp {} /home/data/allData/xml/ \;

# 初始化conda环境（添加此行来解决命令找不到的问题）
conda init bash

# 重新启动shell或者直接执行下一行
source ~/.bashrc

# 激活名为yolopy3.8的conda环境
conda activate yolopy3.8

# 运行Python脚本
python /project/train/src_repo/ultralytics/ultralytics/dataprocess/xmlTyolo.py /home/data/allData/xml/ /home/data/allData/labels/
python /project/train/src_repo/ultralytics/ultralytics/dataprocess/dataSplit.py /home/data/allData/images 0.94 0

# 后台运行yolo检测训练任务，并将输出重定向到日志文件中
nohup yolo detect train data=/project/train/src_repo/ultralytics/ultralytics/datasets/data.yaml model=/project/train/src_repo/ultralytics/ultralytics/cfg/models/v8/yolov8n.yaml epochs=100 imgsz=640 batch=16 device=0 project=/project/train/models epochs=300 > /project/train/log/log.txt &